# 向后兼容的包装器 - 现在使用server.db模块中的功能

# 导入db模块的所有组件
from db import *

# 为了保持向后兼容性，重新导出主要组件
from db.base import engine, SessionLocal, Base, init_db, get_db
from db.models import Category, Document, document_categories
from db.database import Database
from db.vector_store import VectorStore
from db.document_manager import DocumentManager
from db.model_loader import model_loader

# 保持原有的导出结构，确保现有代码可以正常工作
__all__ = [
    'engine', 'SessionLocal', 'Base', 'init_db', 'get_db',
    'Category', 'Document', 'document_categories',
    'Database', 'VectorStore', 'DocumentManager',
    'model_loader',
    'get_db_session', 'get_database', 'get_vector_store', 
    'get_document_manager', 'get_model_loader'
]
